@echo off
echo 正在启动GPU版本OCR识别程序...

REM 检查虚拟环境是否存在
if not exist "venv_gpu" (
    echo 正在创建GPU虚拟环境...
    python -m venv venv_gpu
    call venv_gpu\Scripts\activate
    echo 正在安装GPU版本依赖...
    REM 临时修改requirements.txt以安装GPU版本
    type requirements.txt | findstr /v "paddlepaddle>=" > requirements_gpu.txt
    echo paddlepaddle-gpu>=2.6.1 >> requirements_gpu.txt
    pip install -r requirements_gpu.txt
    del requirements_gpu.txt
) else (
    call venv_gpu\Scripts\activate
)

REM 启动主程序
echo 启动GPU版本OCR识别程序...
python main.py

REM 程序结束后暂停
pause 